package com.ruoyi.ade.llm.model;

import java.util.List;

/**
 * ADE识别配置
 */
public class AdeRecognizeConfig {

    /**
     * 使用哪个大模型
     */
    private String llmModel;

    /**
     *   系统 提示词模板
     */
    private String systemPrompt;

    /**
     *   用户 提示词模板
     */
    private String userPrompt;

    /**
     * 识别ADE后  后置处理
     * eg:
     *   1. 药物实体： 未指定/未知/未知药物 等药物实体替换为空字符串
     *   2. 不良反应实体： 未知/未指定/未知不良反应/空 等不良范围实体替换为空字符串， 如果ADE中reacitons为空 则认为不是药物不良反应事件
     *
     */
    private boolean afterFilter = true;


    public AdeRecognizeConfig(){
        this.afterFilter =  true;
    }

    public AdeRecognizeConfig(String llmModel, String systemPrompt,String userPrompt, boolean afterFilter) {
        this.llmModel = llmModel;
        this.systemPrompt = systemPrompt;
        this.userPrompt = userPrompt;
        this.afterFilter = afterFilter;
    }

    public String getLlmModel() {
        return llmModel;
    }

    public void setLlmModel(String llmModel) {
        this.llmModel = llmModel;
    }

    public String getSystemPrompt() {
        return systemPrompt;
    }

    public void setSystemPrompt(String systemPrompt) {
        this.systemPrompt = systemPrompt;
    }

    public String getUserPrompt() {
        return userPrompt;
    }

    public void setUserPrompt(String userPrompt) {
        this.userPrompt = userPrompt;
    }

    public boolean isAfterFilter() {
        return afterFilter;
    }

    public void setAfterFilter(boolean afterFilter) {
        this.afterFilter = afterFilter;
    }


}
